"""SciPy库本身是针对科学计算而不是图像处理的，只是图像处理也包含了很多数学计算，
所以Scipy也提供了一个专门的模块ndimage用于图像处理。

ndimage模块提供的功能包括输入/输出图像、显示图像、基本操作（如裁剪、翻转、旋转等）、
图像过滤（如去噪、锐化等）、图像分割、分类、特征提取以及注册/配准等任务。

这个模块支持多种图像格式的读取和写入，使得对图像的处理变得方便快捷。

1. 主要功能
虽然图像处理不是Scipy的主要目的，Scipy中也提供了70多个各类图像处理函数。

类别	主要函数	说明
过滤器	包含convolve等20多个函数	各类卷积和滤波相关的计算函数
傅立叶滤波器	包含fourier_ellipsoid等4个函数	多维椭球傅里叶，高斯傅里叶等滤波器
图像插值	包含affine_transform等8个函数	图像的反射变换，移动，旋转等相关函数
图像测量	包含center_of_mass等将近20个函数	计算图像几何特征的相关函数
形态学	包含binary_closing等20多个函数	图像的侵蚀，膨胀，二元开闭运算等等
图像处理底层函数专业性较强，下面结合图片演示一些比较直观的例子。

2. 边缘检测
图像边缘检测在计算机视觉和图像处理中是非常重要的任务之一。
边缘是图像中像素值发生显著变化的地方，它可以提供有关图像的重要信息，例如物体的轮廓、边界等。

ndimage模块中提供了多种算法来检测边缘，下面演示三种不同的边缘检测算法的效果：
（示例中所用的图片是维基百科上找的一个python logo）

2.1. sobel算法"""
import matplotlib.pyplot as plt
import cv2
import matplotlib
from scipy import ndimage
import matplot_config

matplot_config.init_config()

image = plt.imread("../datas/th.jpg")
# 图像灰度化
gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)

# 使用索贝尔边缘检测算法
name = "sobel"
edges = ndimage.sobel(gray)

# 显示原始图像和边缘检测结果
fig, ax = plt.subplots(1, 3, figsize=(8, 4))
ax[0].imshow(image)
ax[0].set_title("原始图像")
ax[1].imshow(gray, cmap="gray")
ax[1].set_title("灰度图像")
ax[2].imshow(edges, cmap="gray")
ax[2].set_title("边缘检测({}算法)".format(name))
plt.show()
"""image.png

2.2. prewitt算法
代码和上面的类似，不同的部分就下面两行。"""

# 使用prewitt边缘检测算法
name = "prewitt"
edges = ndimage.prewitt(gray)
"""image.png

2.3. laplace算法
上面两种算法的效果看上去很类似，laplace算法的结果看上去比上面两种效果更好一些。"""

name = "laplace"
edges = ndimage.laplace(gray)
"""image.png

3. 侵蚀和膨胀
侵蚀和膨胀是最基本的两种图像形态学操作，它们的作用用来增强目标特征。
仍然使用上面的python logo图片，演示侵蚀和膨胀的操作。
"""

image = plt.imread("../datas/th.jpg")
# 图像灰度化
image = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)

# 图像侵蚀
structure = ndimage.generate_binary_structure(2, 2)
erosion = ndimage.binary_erosion(image, structure)

# 图像膨胀
dilation = ndimage.binary_dilation(image, structure)

# 显示原始图像、侵蚀图像和膨胀图像
fig, ax = plt.subplots(1, 3, figsize=(8, 4))
ax[0].imshow(image, cmap="gray")
ax[0].set_title("灰度图像")
ax[1].imshow(erosion, cmap="gray")
ax[1].set_title("图像--侵蚀")
ax[2].imshow(dilation, cmap="gray")
ax[2].set_title("图像--膨胀")
plt.show()
"""image.png
简单来说，侵蚀操作会扩张图像中黑色的区域，反之，膨胀操作会扩张图像中白色的区域。
直观上来看的话，侵蚀变瘦了，膨胀变胖了。

4. 总结
Scipy的图像模块本质上是把图像当作数组来处理，
虽然它不是专门的图像处理库，不过它处理速度很快，且和numpy等库结合紧密，
经常处理图像的朋友可以把它当成一个辅助的工具。"""